package controller

import (
	"fmt"
	"my_first_try/common"
	"my_first_try/model"
	"net/http"
	"strconv"

	"github.com/gin-gonic/gin"
	"gorm.io/gorm"
)

//管理员（用户）登录 (页面用)
func Ad_Login(c *gin.Context) {
	DB := common.GetDB()
	request := make(map[string]interface{}) //新建一个map类型的变量request，保存前端传来的数据
	var test model.Staff
	//获取参数
	c.ShouldBind(&request) //将前端传过来的数据绑定到该request上
	fmt.Println(request)
	test.Staff_id = request["id"].(string) //取得相应的map值 放到结构体上
	test.Staff_password = request["password"].(string)
	fmt.Println("id为")
	fmt.Println(test.Staff_id)
	fmt.Println("密码为")
	fmt.Println(test.Staff_password)

	// 查询管理员是否存在(链接数据库查)
	if Find_Ad(DB, test) {
		fmt.Println("成功查找到该管理员，可以登录")
		c.JSON(200, gin.H{
			"code": "200",
			"msg":  "ok",
		})
		// 开启登录成功后的主页面
	} else {
		c.JSON(200, gin.H{
			"code": "404",
			"msg":  "no",
		})
		fmt.Println("该用户名或密码错误或权限不足，不能登录！")
	}
}

//管理员（用户）登录 (页面用)
func User_Login(c *gin.Context) {
	DB := common.GetDB()
	request := make(map[string]interface{}) //新建一个map类型的变量request，保存前端传来的数据
	var test model.Staff
	//获取参数
	fmt.Println(request)
	c.ShouldBind(&request) //将前端传过来的数据绑定到该request上
	fmt.Println(request)
	test.Staff_id = request["id"].(string) //取得相应的map值 放到结构体上
	test.Staff_password = request["password"].(string)
	fmt.Println("id为")
	fmt.Println(test.Staff_id)
	fmt.Println("密码为")
	fmt.Println(test.Staff_password)

	// 查询管理员是否存在(链接数据库查)
	if Find_Staff(DB, test) {
		fmt.Println("成功查找到该管理员，可以登录")
		c.JSON(200, gin.H{
			"code": "200",
			"msg":  "ok",
		})
		// 开启登录成功后的主页面
	} else {
		c.JSON(200, gin.H{
			"code": "404",
			"msg":  "no",
		})
		fmt.Println("该用户名或密码错误或权限不足，不能登录！")
	}
}

//找到所有员工信息，并传回到网页上 （页面用）
func E_Set_All_Staffs(c *gin.Context) {
	DB := common.GetDB()
	var all_staffs []model.Staff //结构体数组
	// 显示所有的员工信息
	result := DB.Table("staffs").Find(&all_staffs)
	if result != nil {
		fmt.Println("!!")
	}
	fmt.Println(all_staffs)
	c.JSON(http.StatusOK, gin.H{ //多传一些参数 可能会用到
		"code":   "200",
		"staffs": all_staffs, //试试传一个列表
		"msg":    "成功",
	})
}

//查找某一指定员工信息(网页用)
func E_Get_Staff(c *gin.Context) {
	DB := common.GetDB()
	var test model.Staff
	var test2 []model.Staff
	request := make(map[string]interface{}) //新建一个map类型的变量request，保存前端传来的数据
	//获取参数
	fmt.Println(request)
	c.ShouldBind(&request) //将前端传过来的数据绑定到该request上
	fmt.Println(request)
	test.Staff_id = request["id"].(string) //取得相应的map值 放到结构体上
	if test.Staff_id == "" {
		c.JSON(200, gin.H{
			"code":   "404",
			"staffs": test2, //传一个空对象
			"msg":    "no",
		})
	}
	// 查询用户是否存在(链接数据库查)
	DB.Where("staff_id = ?", test.Staff_id).First(&test2)
	if test2 == nil {
		c.JSON(200, gin.H{
			"code":   "404",
			"staffs": test2, //传一个空对象
			"msg":    "no",
		})
	}
	if test2[0].Staff_id != "" {
		fmt.Println("成功查找到该用户")
		c.JSON(200, gin.H{
			"code":   "200",
			"staffs": test2, //传一个对象
			"msg":    "成功",
		})
	} else {
		c.JSON(200, gin.H{
			"code":   "404",
			"staffs": test2, //传一个空对象
			"msg":    "no",
		})
		fmt.Println("未找到该用户")
	}
}

//增加员工（页面用）
func E_Add_Staff(c *gin.Context) {
	DB := common.GetDB()
	request := make(map[string]interface{}) //新建一个map类型的变量request，保存前端传来的数据
	c.ShouldBind(&request)                  //将前端传过来的数据绑定到该request上
	var test model.Staff
	test.Staff_id = request["staff_id"].(string)             //获取前端的要增加的员工的属性
	test.Staff_name = request["staff_name"].(string)         //获得姓名
	test.Staff_password = request["staff_password"].(string) //获得密码
	test.Staff_telephone = request["staff_telephone"].(string)
	test.Staff_level = request["staff_level"].(string)
	test.Staff_remarks = request["staff_remarks"].(string)
	test.Staff_salary, _ = strconv.Atoi(request["staff_salary"].(string))
	fmt.Println(request)
	//获取到新增员工的所有信息
	if Insert_Staff(DB, test) {
		//没找到，可以新建该员工
		c.JSON(200, gin.H{
			"code": "200",
			"msg":  "添加成功",
		})
		fmt.Println("添加成功")
	} else {
		//能找到，就不可以新建该员工
		c.JSON(200, gin.H{
			"code": "404",
			"msg":  "添加失败",
		})
		fmt.Println("添加失败")
	}
}

//删除指定员工（页面用）
func E_Delete_Staff(c *gin.Context) {
	DB := common.GetDB()
	request := make(map[string]interface{}) //新建一个map类型的变量request，保存前端传来的数据
	c.ShouldBind(&request)                  //将前端传过来的数据绑定到该request上
	var test model.Staff
	test.Staff_id = request["staff_id"].(string) //获取前端的要删除的员工的id
	if Delete_Staff(DB, test) {
		//删除成功
		c.JSON(200, gin.H{
			"code": "200",
			"msg":  "删除成功",
		})
		fmt.Println("删除成功")
	} else {
		//删除失败
		c.JSON(200, gin.H{
			"code": "404",
			"msg":  "删除失败",
		})
		fmt.Println("删除失败")
	}
}

//修改指定员工信息（页面用）
func E_Modify_Staff(c *gin.Context) {
	DB := common.GetDB()
	request := make(map[string]interface{}) //新建一个map类型的变量request，保存前端传来的数据
	c.ShouldBind(&request)                  //将前端传过来的数据绑定到该request上
	var test model.Staff
	test.Staff_id = request["staff_id"].(string)             //获取前端的要增加的员工的属性
	test.Staff_name = request["staff_name"].(string)         //获得姓名
	test.Staff_password = request["staff_password"].(string) //获得密码
	test.Staff_telephone = request["staff_telephone"].(string)
	test.Staff_level = request["staff_level"].(string)
	test.Staff_remarks = request["staff_remarks"].(string)
	test.Staff_salary, _ = strconv.Atoi(request["staff_salary"].(string))
	//获取到全部修改信息

	if Modify_Staff(DB, test) {
		c.JSON(200, gin.H{
			"code": "200",
			"msg":  "删除成功",
		})
	} else {
		c.JSON(200, gin.H{
			"code": "404",
			"msg":  "删除失败",
		})
	}
}

//管理员级别
//查询管理员是否存在（登陆用）

func Find_Ad(db *gorm.DB, test model.Staff) bool {
	var staff model.Staff
	db.Where("staff_id = ? AND staff_password = ? AND staff_level = '0'", test.Staff_id, test.Staff_password).First(&staff)
	if staff.Staff_id != "" {
		fmt.Println(staff.Staff_name)
		return true
	}
	return false
}

//员工登录
func Find_Staff(db *gorm.DB, test model.Staff) bool {
	var staff model.Staff
	db.Where("staff_id = ? AND staff_password = ?", test.Staff_id, test.Staff_password).First(&staff)
	if staff.Staff_id != "" {
		fmt.Println(staff.Staff_name)
		return true
	}
	return false
}

//查询员工是否存在（新建用户用）
func Find_Staff_2(db *gorm.DB, test model.Staff) bool {
	var staff model.Staff
	db.Where("staff_id = ?", test.Staff_id).First(&staff)
	if staff.Staff_id != "" {
		fmt.Println(staff.Staff_name)
		return true
	}
	return false
}

// 插入员工
func Insert_Staff(db *gorm.DB, test model.Staff) bool {
	if Find_Staff_2(db, test) {
		fmt.Println("该用户已经存在了！不能插入!")
		return false
	} else {
		db.Create(&test)
		fmt.Println("插入成功")
		return true
	}
}

// 删除员工
func Delete_Staff(db *gorm.DB, test model.Staff) bool {
	if Find_Staff_2(db, test) {
		fmt.Println("已经查找到该用户")
		db.Where("staff_id = ?", test.Staff_id).Delete(&test)
		return true
	} else {
		fmt.Println("并不存在该用户")
	}
	return false
}

// 修改员工
func Modify_Staff(db *gorm.DB, test model.Staff) bool {
	if Find_Staff_2(db, test) {
		fmt.Println("已经查找到该用户")
		db.Model(&test).Where("staff_id = ?", test.Staff_id).Updates(
			model.Staff{
				Staff_id:        test.Staff_id,
				Staff_name:      test.Staff_name,
				Staff_password:  test.Staff_password,
				Staff_level:     test.Staff_level,
				Staff_telephone: test.Staff_telephone,
				Staff_salary:    test.Staff_salary,
				Staff_remarks:   test.Staff_remarks,
			},
		)
		fmt.Println("修改完成")
		return true
	} else {
		fmt.Println("并不存在该用户")
	}
	return false
}
